Mean variance and standard deviation in python